"""使用Pandas读取Excel文件中的数据"""
import pandas as pd

# 设置列对齐
pd.set_option('display.unicode.ambiguous_as_wide', True)
pd.set_option('display.unicode.east_asian_width', True)

# 读取工号、姓名、时段、交易额这四列数据，使用默认索引
# 如果可以确定数据类型，使用usecols={'工号':str,'姓名':str,'交易额':float}可以提高速度
df = pd.read_excel(r'resources/超市营业额2.xlsx',
                   usecols=['工号', '姓名', '时段', '交易额'])
# 输出前10行数据
print(df[:10], end='\n\n')

# 读取第一个worksheet中所有列
# 跳过第1、3、5行，指定下标为1的列中数据为DataFrame的行索引标签
df = pd.read_excel(r'resources/超市营业额2.xlsx',
                   skiprows=[1, 3, 5],
                   index_col=1)
# 输出前10行数据
print(df[:10])
